//
//  UncaughtExceptionHandler.m
//  UncaughtExceptions
//
//  Created by Matt Gallagher on 2010/05/25.
//  Copyright 2010 Matt Gallagher. All rights reserved.
//
//  Permission is given to use this source code file, free of charge, in any
//  project, commercial or otherwise, entirely at your risk, with the condition
//  that any redistribution (in part or whole) of source code must retain
//  this copyright and permission notice. Attribution in compiled projects is
//  appreciated but not required.
//

#import "UncaughtExceptionHandler.h"
#include <libkern/OSAtomic.h>
#include <execinfo.h>

NSString * const UncaughtExceptionHandlerSignalExceptionName = @"UncaughtExceptionHandlerSignalExceptionName";
NSString * const UncaughtExceptionHandlerSignalKey = @"UncaughtExceptionHandlerSignalKey";
NSString * const UncaughtExceptionHandlerAddressesKey = @"UncaughtExceptionHandlerAddressesKey";

volatile int32_t UncaughtExceptionCount = 0;
const int32_t UncaughtExceptionMaximum = 10;

const NSInteger UncaughtExceptionHandlerSkipAddressCount = 4;
const NSInteger UncaughtExceptionHandlerReportAddressCount = 5;

static ExceptionInfoWillThrow s_throw;

@implementation UncaughtExceptionHandler

+ (NSArray *)backtrace
{
	 void* callstack[128];
	 int frames = backtrace(callstack, 128);
	 char **strs = backtrace_symbols(callstack, frames);
	 
	 int i;
	 NSMutableArray *backtrace = [NSMutableArray arrayWithCapacity:frames];
	 for (
	 	i = UncaughtExceptionHandlerSkipAddressCount;
	 	i < UncaughtExceptionHandlerSkipAddressCount +
			UncaughtExceptionHandlerReportAddressCount;
		i++)
	 {
	 	[backtrace addObject:[NSString stringWithUTF8String:strs[i]]];
	 }
	 free(strs);
	 
	 return backtrace;
}

- (void)alertView:(UIAlertView *)anAlertView clickedButtonAtIndex:(NSInteger)anIndex
{
//	if (anIndex == 0)
	{
		dismissed = YES;
	}
}

- (void)handleException:(NSException *)exception
{
//	UIAlertView *alert =
//		[[[UIAlertView alloc]
//			initWithTitle:NSLocalizedString(@"Unhandled exception", nil)
//			message:[NSString stringWithFormat:NSLocalizedString(
//				@"You can try to continue but the application may be unstable.\n\n"
//				@"Debug details follow:\n%@\n%@", nil),
//				[exception reason],
//				[[exception userInfo] objectForKey:UncaughtExceptionHandlerAddressesKey]]
//			delegate:self
//			cancelButtonTitle:NSLocalizedString(@"Quit", nil)
//			otherButtonTitles:NSLocalizedString(@"Continue", nil), nil]
//		autorelease];
//	[alert show];
    
    if(s_throw!=NULL)
        s_throw(exception);
	
    NSDictionary* dicInfo=[[NSBundle mainBundle] infoDictionary];
    UIAlertView *alert = [[[UIAlertView alloc]
      initWithTitle:@"异常错误"                                                                                                       
                           message:[NSString stringWithFormat:@"“%@ %@ %@”发生异常错误！",[dicInfo objectForKey:@"CFBundleDisplayName"],[dicInfo objectForKey:@"CFBundleShortVersionString"],[dicInfo objectForKey:@"CFBundleVersion"]]
      delegate:self
      cancelButtonTitle:@"退出"
      otherButtonTitles:nil]
     autorelease];
	[alert show];
    
	CFRunLoopRef runLoop = CFRunLoopGetCurrent();
	CFArrayRef allModes = CFRunLoopCopyAllModes(runLoop);
	
	while (!dismissed)
	{
		for (NSString *mode in (NSArray *)allModes)
		{
			CFRunLoopRunInMode((CFStringRef)mode, 0.001, false);
		}
	}
	
	CFRelease(allModes);

	UninstallUncaughtExceptionHandler();
    
	if ([[exception name] isEqual:UncaughtExceptionHandlerSignalExceptionName])
	{
		kill(getpid(), ToInt [[[exception userInfo] objectForKey:UncaughtExceptionHandlerSignalKey] integerValue]);
	}
	else
	{
		[exception raise];
	}
}

@end

void HandleException(NSException *exception)
{
	int32_t exceptionCount = OSAtomicIncrement32(&UncaughtExceptionCount);
	if (exceptionCount > UncaughtExceptionMaximum)
	{
		return;
	}
	
//	NSArray *callStack = [UncaughtExceptionHandler backtrace];
    NSArray *callStack = [exception callStackSymbols];
	NSMutableDictionary *userInfo = [NSMutableDictionary dictionaryWithDictionary:[exception userInfo]];
	[userInfo setObject:callStack forKey:UncaughtExceptionHandlerAddressesKey];
	
	[[[[UncaughtExceptionHandler alloc] init] autorelease]
		performSelectorOnMainThread:@selector(handleException:)
		withObject: [NSException exceptionWithName:[exception name] reason:[exception reason] userInfo:userInfo]
		waitUntilDone:YES];
}

void SignalHandler(int signal)
{
	int32_t exceptionCount = OSAtomicIncrement32(&UncaughtExceptionCount);
	if (exceptionCount > UncaughtExceptionMaximum)
	{
		return;
	}
	
	NSMutableDictionary *userInfo =
		[NSMutableDictionary
			dictionaryWithObject:[NSNumber numberWithInt:signal]
			forKey:UncaughtExceptionHandlerSignalKey];

	NSArray *callStack = [UncaughtExceptionHandler backtrace];
	[userInfo
		setObject:callStack
		forKey:UncaughtExceptionHandlerAddressesKey];
	
	[[[[UncaughtExceptionHandler alloc] init] autorelease]
		performSelectorOnMainThread:@selector(handleException:)
		withObject: [NSException exceptionWithName:UncaughtExceptionHandlerSignalExceptionName
				reason: [NSString stringWithFormat: NSLocalizedString(@"Signal %d was raised.", nil), signal]
				userInfo: [NSDictionary dictionaryWithObject:[NSNumber numberWithInt:signal] forKey:UncaughtExceptionHandlerSignalKey]]
		waitUntilDone:YES];
}

void InstallUncaughtExceptionHandler(ExceptionInfoWillThrow expthrow)
{
    if(expthrow!=NULL)
        s_throw=Block_copy(expthrow);
	NSSetUncaughtExceptionHandler(&HandleException);
	signal(SIGABRT, SignalHandler);
	signal(SIGILL, SignalHandler);
	signal(SIGSEGV, SignalHandler);
	signal(SIGFPE, SignalHandler);
	signal(SIGBUS, SignalHandler);
	signal(SIGPIPE, SignalHandler);
}

void UninstallUncaughtExceptionHandler()
{
    if(s_throw!=NULL)
        Block_release(s_throw);
    s_throw=NULL;
	NSSetUncaughtExceptionHandler(NULL);
	signal(SIGABRT, SIG_DFL);
	signal(SIGILL, SIG_DFL);
	signal(SIGSEGV, SIG_DFL);
	signal(SIGFPE, SIG_DFL);
	signal(SIGBUS, SIG_DFL);
	signal(SIGPIPE, SIG_DFL);
}

NSString* ExceptionInfoDefaultFormat(NSException* exception)
{
    NSMutableString* s=[[NSMutableString new]autorelease];
    
    [s appendString:@"------------------------"];
    [s appendString:@"\n"];
    
    NSDictionary* info=[[NSBundle mainBundle]infoDictionary];
    [s appendFormat:@"%@ %@ %@",[info objectForKey:@"CFBundleDisplayName"],[info objectForKey:@"CFBundleShortVersionString"],[info objectForKey:@"CFBundleVersion"]];
    [s appendString:@"\n"];
    
    [s appendString:[exception name].length>0?[exception name]:@""];
    [s appendString:@"\n"];
    
    [s appendString:[exception reason].length>0?[exception reason]:@""];
    [s appendString:@"\n"];
    
    [s appendString:[(NSArray*)[[exception userInfo] objectForKey:UncaughtExceptionHandlerAddressesKey] description]];
    
    return s;
}

